package cn.iocoder.yudao.module.ai.dal.dataobject.chat;

import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiChatRoleDO;
import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiModelDO;
import com.baomidou.mybatisplus.annotation.KeySequence;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.*;

import java.time.LocalDateTime;

/**
 * AI Chat 对话 DO
 *
 * 用户每次发起 Chat 聊天时，会创建一个 {@link AiChatConversationDO} 对象，将它的消息关联在一起
 *
 * @author fansili
 * @since 2024/4/14 17:35
 */
@TableName("ai_chat_conversation")
@KeySequence("ai_chat_conversation_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class AiChatConversationDO extends BaseDO {

    public static final String TITLE_DEFAULT = "新对话";

    /**
     * ID 编号，自增
     */
    @TableId
    private Long id;

    /**
     * 用户编号
     *
     * 关联 AdminUserDO 的 userId 字段
     */
    private Long userId;

    /**
     * 对话标题
     *
     * 默认由系统自动生成，可用户手动修改
     */
    private String title;
    /**
     * 是否置顶
     */
    private Boolean pinned;
    /**
     * 置顶时间
     */
    private LocalDateTime pinnedTime;

    /**
     * 角色编号
     *
     * 关联 {@link AiChatRoleDO#getId()}
     */
    private Long roleId;

    /**
     * 模型编号
     *
     * 关联 {@link AiModelDO#getId()} 字段
     */
    private Long modelId;
    /**
     * 模型标志
     *
     * 冗余 {@link AiModelDO#getModel()} 字段
     */
    private String model;

    // ========== 对话配置 ==========

    /**
     * 角色设定
     */
    private String systemMessage;
    /**
     * 温度参数
     *
     * 用于调整生成回复的随机性和多样性程度：较低的温度值会使输出更收敛于高频词汇，较高的则增加多样性
     */
    private Double temperature;
    /**
     * 单条回复的最大 Token 数量
     */
    private Integer maxTokens;
    /**
     * 上下文的最大 Message 数量
     */
    private Integer maxContexts;



//    CREATE TABLE ai_chat_conversation (
//            id BIGINT NOT NULL AUTO_INCREMENT COMMENT 'ID 编号，自增',
//            user_id BIGINT NOT NULL COMMENT '用户编号，关联 AdminUserDO 的 userId 字段',
//            title VARCHAR(255) DEFAULT '新对话' COMMENT '对话标题，默认由系统自动生成，可用户手动修改',
//    pinned TINYINT(1) DEFAULT 0 COMMENT '是否置顶',
//    pinned_time DATETIME DEFAULT NULL COMMENT '置顶时间',
//    role_id BIGINT DEFAULT NULL COMMENT '角色编号，关联 AiChatRoleDO 的 id 字段',
//    model_id BIGINT DEFAULT NULL COMMENT '模型编号，关联 AiModelDO 的 id 字段',
//    model VARCHAR(255) DEFAULT NULL COMMENT '模型标志，冗余 AiModelDO 的 model 字段',
//    system_message TEXT COMMENT '角色设定',
//    temperature DOUBLE DEFAULT NULL COMMENT '温度参数，用于调整生成回复的随机性和多样性程度',
//    max_tokens INT DEFAULT NULL COMMENT '单条回复的最大 Token 数量',
//    max_contexts INT DEFAULT NULL COMMENT '上下文的最大 Message 数量',
//
//            -- BaseDO 字段
//    creator BIGINT DEFAULT NULL COMMENT '创建者',
//    create_time DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
//    updater BIGINT DEFAULT NULL COMMENT '更新者',
//    update_time DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',
//    deleted TINYINT(1) DEFAULT 0 COMMENT '删除标识：0-未删除；1-已删除',
//    tenant_id BIGINT DEFAULT NULL COMMENT '租户编号',
//
//    PRIMARY KEY (id)
//) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='AI Chat 对话'
//

}
